/*
 * To change this template, choose Tools | Templates
 * and open the template in the editor.
 */
package grex.WekaModels;

import grex.WekaModels.WekaPredictiveModel;
import grex.Data.ArffTableModel;
import grex.Environment;
import weka.classifiers.trees.J48;
import weka.classifiers.trees.RandomForest;
import weka.core.Instance;
import weka.core.Instances;

/**
 *
 * @author RIK
 */
public class GrexRandomForest extends WekaPredictiveModel {
    public RandomForest forest;
    int nrOfTrees = 20;
    public GrexRandomForest(ArffTableModel data){
        super(data, new RandomForest());
        forest = (RandomForest)model;
        forest.setNumTrees(nrOfTrees);
        
    }
    
        public GrexRandomForest(Environment env){
        super(env,new RandomForest());
        forest = (RandomForest)model;
        forest.setNumTrees(nrOfTrees);
    }
    
    public double getNrOfNodes() {

        return nrOfTrees;
    }

    public String getName() {
        return "RF";
    }
    
}
